Pattern program in python